package com.wk.util;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.log4j.Logger;

/**
 * 压缩工具类
 * @author chenxiaozhong
 *
 */
public class FileToZip {
    
    public static Logger logger=Logger.getLogger(FileToZip.class);
    /**
     * 
     * @param sourceFilePath 待压缩的文件路径
     * @param zipFilePath 压缩后存放路径
     * @param fileName 压缩后文件的名称
     * @return
     */
    public static boolean tozip(String sourceFilePath, String zipFilePath, String fileName) {
        
        boolean flag = false;
        File sourceFile = new File(sourceFilePath);
        FileInputStream fis = null;
        BufferedInputStream bis = null;
        FileOutputStream fos = null;
        ZipOutputStream zos = null;

        if (sourceFile.exists() == false) {
            logger.info(sourceFilePath + "没有需要压缩的文件");
        }  
        else 
        {
            try {
                File zipFile = new File(zipFilePath + "/" + fileName + ".zip");
                if (zipFile.exists()) {
                    logger.info(sourceFilePath + "已经存在" + fileName + ".zip");
                }
                else 
                {
                    File[] sourceFiles = sourceFile.listFiles();
                    if (null == sourceFiles || sourceFiles.length < 1) {
                        logger.info(sourceFilePath + "里面不存在文件，无需压缩.");
                    }
                    else 
                    {
                        fos = new FileOutputStream(zipFile);
                        zos = new ZipOutputStream(new BufferedOutputStream(fos));
                        byte[] bufs = new byte[1024 * 10];
                        for (int i = 0; i < sourceFiles.length; i++) {
                            // 创建ZIP实体，并添加进压缩包
                            ZipEntry zipEntry = new ZipEntry( sourceFiles[i].getName());
                            System.out.println(zipEntry);
                            // encoding
                            zos.putNextEntry(zipEntry);
                            // 读取待压缩的文件并写进压缩包里
                            fis = new FileInputStream(sourceFiles[i]);
                            bis = new BufferedInputStream(fis, 1024 * 10);
                            int read = 0;
                            while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                                zos.write(bufs, 0, read);
                            }
                            
                            fis.close();
                        }
                        fos.close();
                        flag = true;
                    }
                }
            }
            catch (FileNotFoundException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            }
            catch (IOException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            }
            finally {
                try {
                    if (null != bis)
                        bis.close();
                    /*if (null != zos)
                        zos.close();*/
                }
                catch (IOException e) {
                    e.printStackTrace();
                    throw new RuntimeException(e);
                }
            }
        }
        return flag;

    }
}
